from data.animehint_dataset import AnimehintDataset
from models.networks.hint_model import HintModel
from option import Options
import matplotlib.pyplot as plt

from utils.util import toNormalPic

if __name__ == '__main__':
    train_parser = Options()
    opt = train_parser.parse()
    # model = HintModel(opt)

    dataset = AnimehintDataset(opt)
    test = dataset.__getitem__(0)
    sketch = test['sketch']
    img = test['image']
    hints = test['hints']

    img = toNormalPic(img)
    hints = toNormalPic(hints)
    sketch = toNormalPic(sketch)
    plt.subplot(131)
    plt.imshow(img)
    plt.subplot(132)
    plt.imshow(sketch)
    plt.subplot(133)
    plt.imshow(hints)
    plt.show()